package com.llk.rabbitmqlearn.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * 自定义消息确认回调接口
 */
@Slf4j
@Component
public class MyConfirmCallback implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnCallback {

    /**
     * 将现在的实现类，注入到 RabbitTemplate.ConfirmCallback
     * 执行顺序 -> 1.类的实例化 -> 2.将RabbitTemplate注入到本类 -> 3.将实现的ConfirmCallback方法注入到RabbitTemplate
     * 下边的注入顺序不能变
     */
    @Autowired
    private RabbitTemplate rabbitTemplate;

    @PostConstruct
    public void init() {
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     * 交换机确认回调方法
     *
     * @param correlationData 回调消息的ID及相关信息
     * @param ack 交换机是否收到消息
     * @param cause 失败原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        if (ack) {
            log.info("交换机成功收到消息，消息ID为：{}", correlationData.getId());
        } else {
            log.info("交换机未收到消息，消息ID为：{}，原因为：{}", correlationData.getId(), cause);
        }
    }

    /**
     * 当消息发送过程中，不可达目的地时，将消息返回给生产者
     * 只有不可达，才会调用（发送失败）
     *
     * @param message
     * @param replyCode
     * @param replyText
     * @param exchange
     * @param routingKey
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.error("消息被交换机退回，消息={}，交换机={}，路由键={}，原因为={}", message.getBody(), exchange, routingKey, replyText);
    }
}
